xHC: Expanded Hyper-Connections
扩展超连接:突破流数瓶颈实现高效大模型预训练
🧭 LLM架构优化 · 残差流扩展
School of AI, Shanghai Jiao Tong University · Dots Studio, Xiaohongshu Inc. · University of Science and Technology of China · School of CS, Peking University · The Chinese University of Hong Kong
通过时间特征增强和稀疏残差流(仅更新k=4/N=16)突破HC的N=4瓶颈,在MoE模型上大幅提升性能并降低计算开销
📌 工作参考价值(4 点)
- 稀疏更新策略:xHC 每层只更新 k=4 条流但读取全部流,类似推荐系统中大规模 Embedding 表或状态存储可采用“稠密读取-稀疏更新”模式,仅刷新部分高频/重要行,降低训练更新开销。
- 多流残差结构:多条并行残差流可视为不同变换路径的集成,该思想可迁移到推荐模型的多模态、多任务学习,用不同流编码用户、商品、上下文等异构特征,再通过可学习的混合机制融合。
- 内存带宽优化:xHC-Flash 通过融合流维度与批次维度的矩阵乘法减少中间状态传输,对推荐模型部署中的显存带宽瓶颈有借鉴意义,尤其适用于大规模特征交叉或 Embedding 交互层的显存优化。
- 高效扩展范式:在有限训练预算下通过架构改进提升大模型性能,对电商搜索/推荐场景中自行预训练 LLM 的团队,提供了一条除增加模型尺寸外的算力友好扩展路径。